# -*- coding:utf-8 -*- 
# Author: json_steve
# -*- coding:utf-8 -*-

# 实现注册和登录的逻辑

from flask import current_app,session
from . import api
from flask import request, jsonify
from iHome.utils.response_code import RET
from iHome import redis_store, db
from iHome.models import User


# 注册路由 短信验证
@api.route("/users", methods=["POST"])
def register():
    """
    1. 获取参数和判断是否有值
    2. 从redis中获取指定手机号对应的短信验证码的
    3. 校验验证码
    4. 初始化 user 模型，并设置数据并添加到数据库
    5. 保存当前用户的状态
    6. 返回注册的结果
    :return:
    """

    # 1. 获取参数和判断是否有值
    data_dict = request.json
    mobile = data_dict.get("mobile")
    phonecode = data_dict.get("phonecode")
    password = data_dict.get("password")

    if not all([mobile, phonecode, password]):
        return jsonify(errno=RET.PARAMERR, errmsg="参数不全")

    # 2. 从redis中获取指定手机号对应的短信验证码的
    try:
        sms_code = redis_store.get("SMS_"+mobile)
    except Exception as e:
        current_app.logger.error(e)
        return jsonify(errno=RET.DBERR, errmsg="获取本地验证码失败")

    if not sms_code:
        return jsonify(errno=RET.NODATA, errmsg="短信验证码过期")

    # 3. 校验验证码
    if phonecode != sms_code:
        return jsonify(errno=RET.DATAERR, errmsg="短信验证码错误")

    # 4. 初始化 user 模型，并设置数据并添加到数据库
    user = User()
    user.name = mobile
    user.mobile = mobile
    # TODO: 对密码进行处理
    user.password=password

    try:
        db.session.add(user)
        db.session.commit()
    except Exception as e:
        db.session.rollback()
        current_app.logger.error(e)
        return jsonify(errno=RET.DATAERR, errmsg="数据保存错误")
    session["user_id"] = user.id
    session["name"] = user.name
    session["mobile"] = user.mobile

    return jsonify(errno=RET.OK, errmsg="注册成功")


# 登陆路由
@api.route("/session", methods=["POST"])
def login():
    """
    1. 获取参数和判断是否有值
    2. 从数据库查询出指定的用户
    3. 校验密码
    4. 保存用户登录状态
    5. 返回结果
    :return:
    """

    # 1. 获取参数和判断是否有值
    data_dict = request.json

    mobile = data_dict.get("mobile")
    password = data_dict.get("password")
    print type(mobile)
    if not all([mobile, password]):
        return jsonify(errno=RET.PARAMERR, errmsg="参数不全")

    # 2. 从数据库查询出指定的用户
    try:
        user = User.query.filter_by(mobile=mobile).first()  # unicode 字符串不识别？？
    except Exception as e:
        current_app.logger.error(e)
        return jsonify(errno=RET.DBERR, errmsg="查询数据错误")

    if not user:
        return jsonify(errno=RET.USERERR, errmsg="用户不存在")

    # 3. 校验密码
    if not user.check_passowrd(password):
        return jsonify(errno=RET.PWDERR, errmsg="密码错误")

    # 4. 保存用户登录状态
    session["user_id"] = user.id
    session["mobile"] = user.mobile
    session["name"] = user.name

    return jsonify(errno=RET.OK, errmsg="登录成功")


# 推出登陆
@api.route('/session', methods=['delete'])
def logout():
    """
    1.清除session中保存的用户信息，user_id,name,mobile
    :return:
    """
    session.pop('user_id',None)
    session.pop('name',None)
    session.pop('mobile',None)
    return jsonify(errno=RET.OK,errmsg='ok')